package com.photovoltaic.neo4j.entity;

import lombok.Data;
import java.util.List;

@Data
public class Company {
    private String id;
    private String name;                // 公司名称
    private Integer employeeCount;      // 员工数量
    private String companyScale;       // 企业规模
    private Double registeredCapital;  // 注册资本
    private Integer establishYear;      // 成立年份
    private String province;           // 省市
    private Integer riskCount;         // 风险数量
    private List<String> riskTypes;
    private Boolean isHighTech;        // 是否高新企业

    // 必须的setter方法
    public void setId(String id) {
        this.id = id;  // 直接使用字符串 id
    }
    public void setName(String name) { this.name = name; }
    public void setRiskTypes(List<String> riskTypes) {
        this.riskTypes = riskTypes;
    }

    public List<String> getRiskTypes() {
        return riskTypes;
    }

    public void setHighTech(Boolean isHighTech) {
        this.isHighTech = isHighTech;
    }

    public Boolean getHighTech() {
        return isHighTech;
    }
} 